model_name = "FEAREC"
dataset_name = "taobao"
seed = 42  # 设定一个随机种子
n_users = 512

BATCH_SIZE = 512
EPOCHS = 10
LEARNING_RATE = 1e-4
WEIGHT_DECAY = 1e-5
heads=8
k=50
group_num = 100
interest_num=4
short_time=20
if dataset_name == "xlong":
    total_behavior_num=1000
    id_embedding_dim=16
else:
    total_behavior_num=200
    id_embedding_dim=64
simple_embedding_dim=8
layer = 4
drop_out=0.2
short_time = 20
device_id = 0
sim_mode = "soft"

split_range = [0.20, 0.30, 0.50]
